package beautiful.butterfly.drds.data_exchange.core.statistics.plugin;


import beautiful.butterfly.drds.data_exchange.core.statistics.communication.Communication;
import beautiful.butterfly.drds.data_exchange.core.statistics.container.communicator.AbstractContainerCommunicator;
import common.plugin.JobPluginCollector;

import java.util.List;
import java.util.Map;


public final class DefaultJobPluginCollector implements JobPluginCollector
{
    private AbstractContainerCommunicator jobCollector;

    public DefaultJobPluginCollector(AbstractContainerCommunicator containerCollector)
    {
        this.jobCollector = containerCollector;
    }


    public Map<String, List<String>> getMessage()
    {
        Communication totalCommunication = this.jobCollector.collect();
        return totalCommunication.getMessage();
    }


    public List<String> getMessage(String key)
    {
        Communication totalCommunication = this.jobCollector.collect();
        return totalCommunication.getMessage(key);
    }
}
